package test;


import java.util.Base64;

import javax.sql.DataSource;

import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;
import com.jfinal.plugin.druid.DruidPlugin;

public class TCXianrenGenerate {

	public static DataSource getDataSource() {
		// PropKit.use("config.txt");
		// C3p0Plugin c3p0Plugin = new C3p0Plugin(PropKit.get("jdbcUrl").trim(),
		// PropKit.get("user").trim(),
		// PropKit.get("password").trim());

		Prop p = PropKit.use("config_dev.properties");
		String password = new String(Base64.getDecoder().decode(p.get("password").trim()));

		DruidPlugin dp = new DruidPlugin(p.get("jdbcUrl"), p.get("user"), password);
		dp.start();

		return dp.getDataSource();
	}

	public static void main(String[] args) throws Exception {
		// base model 所使用的包名
		String baseModelPackageName = "com.rfchina.okr.xianren.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = "src/main/java/com/rfchina/okr/xianren/model/base/";
		System.out.println(baseModelOutputDir);
		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "com.rfchina.okr.xianren.model";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";
		System.out.println(modelOutputDir);
		// 创建生成器
		Generator gernerator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir,
				modelPackageName, modelOutputDir);
		// 添加不需要生成的表名
		gernerator.addExcludedTable("adv");
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// gernerator.setDataDictionaryFileName("microboss.txt");
		// gernerator.setDataDictionaryOutputDir("docs");
		// 设置是否生成字典文件
		// gernerator.setGenerateDataDictionary(true);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为
		// "User"而非 OscUser
		// gernerator.setRemovedTableNamePrefixes("pf_");

		// 生成
		gernerator.generate();
	}

}
